from langchain_openai import ChatOpenAI

from ai_configs import default_ai_configs

_ai_config = default_ai_configs['huoshan-doubao']

# 初始化模型
_llm = ChatOpenAI(
  base_url=_ai_config["url"].replace("/chat/completions", ""),
  api_key=_ai_config["key"],
  model=_ai_config["model"],
)

# 使用模型
_response = _llm.invoke("你好，你可以做什么")
print(_response.content)
